package com.xn2001.controller;

import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletRequest;
import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.UUID;

/**
 * Created by 乐心湖 on 2020/4/17 22:21
 */
@RestController
public class FileUploadController {

    @PostMapping("/upload")
    public String upload(MultipartFile file, HttpServletRequest request) {

        //使用格式化的日期，不过我这里并没有使用到，一般时间可以用来文件分类，文件取名
        SimpleDateFormat format = new SimpleDateFormat("/yyyy/MM/dd");
        String date = format.format(new Date());

        //获取项目所在服务器的全路径
        String realPath = request.getServletContext().getRealPath("/uploadFile");
        System.out.println(realPath);

        File saveFile = new File(realPath);
        if (!saveFile.exists()) {
            saveFile.mkdirs();
        }
        String oldNmae = file.getOriginalFilename();

        //使用UUID随机生成文件名，相当于重命名，你也可以使用原来的文件名, 也可以使用上面的格式化日期。
        String newName = UUID.randomUUID().toString() + oldNmae.substring(oldNmae.lastIndexOf("."));

        try {
            file.transferTo(new File(saveFile, newName));
            String url = request.getScheme() + "://" + request.getServerName() + ":" + request.getServerPort() + "/uploadFile/" + newName;
            return url;
        } catch (IOException e) {
            return "上传失败，异常：" + e.getCause();
        }

    }

    @PostMapping("/uploads")
    public String uploads(MultipartFile[] files, HttpServletRequest request) {

        //使用格式化的日期，不过我这里并没有使用到，一般时间可以用来文件分类，文件取名
        SimpleDateFormat format = new SimpleDateFormat("/yyyy/MM/dd");
        String date = format.format(new Date());

        //获取项目所在服务器的全路径
        String realPath = request.getServletContext().getRealPath("/uploadFile");
        System.out.println(realPath);

        File saveFile = new File(realPath);
        if (!saveFile.exists()) {
            saveFile.mkdirs();
        }

        for (MultipartFile file : files) {

            String oldNmae = file.getOriginalFilename();

            //使用UUID随机生成文件名，相当于重命名，你也可以使用原来的文件名, 也可以使用上面的格式化日期。
            String newName = UUID.randomUUID().toString() + oldNmae.substring(oldNmae.lastIndexOf("."));

            try {
                file.transferTo(new File(saveFile, newName));
                String url = request.getScheme() + "://" + request.getServerName() + ":" + request.getServerPort() + "/uploadFile/" + newName;
                System.out.println(url);
            } catch (IOException e) {
                return "error";
            }
        }
        return "上传成功!";
    }
}
